package com.tanhua.dubbo.server.service;

import cn.hutool.core.collection.CollUtil;
import com.tanhua.dubbo.server.pojo.TimeLine;
import com.tanhua.dubbo.server.pojo.Users;
import lombok.extern.slf4j.Slf4j;
import org.bson.types.ObjectId;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.concurrent.CompletableFuture;


@Service
@Slf4j
public class TimeLineService {

    @Autowired
    private MongoTemplate mongoTemplate;

    @Async//异步执行，原理:底层开一个线程去执行该方法
    public CompletableFuture<String> saveTimeLine(Long userId, ObjectId publishId){
        //写入好友的时间线表

        try {
            //查询好友列表，在MongoDB中的tanhua-users中查询自己的好友
            Query query=Query.query(Criteria.where("userId").is(userId));
            //得到自己的好友集合
            List<Users> usersList = mongoTemplate.find(query, Users.class);
            //判断集合是否为空
            if (CollUtil.isEmpty(usersList)){
                //因为没有好友，直接返回
                return CompletableFuture.completedFuture("ok");
            }

            //将自己的动态加入自己的时间线表
            TimeLine myTimeLine=new TimeLine();
            myTimeLine.setId(ObjectId.get());
            myTimeLine.setDate(System.currentTimeMillis());
            myTimeLine.setPublishId(publishId);
            myTimeLine.setUserId(userId);

            //写入到时间线表（时间线表就是可以看到自己的所有好友发布的动态的简要信息，是简要信息）
            mongoTemplate.save(myTimeLine,"quanzi_time_line_"+userId);


            //遍历好友集合，并写入到时间线表中
            for (Users users : usersList) {
                TimeLine timeLine=new TimeLine();
                timeLine.setId(ObjectId.get());
                timeLine.setDate(System.currentTimeMillis());
                timeLine.setPublishId(publishId);
                timeLine.setUserId(userId);

                //写入到时间线表（时间线表就是可以看到自己的所有好友发布的动态的简要信息，是简要信息）
                mongoTemplate.save(timeLine,"quanzi_time_line_"+users.getFriendId());
            }
        } catch (Exception e) {
            log.error("写入好友时间线表失败！ userId="+userId+",publishId="+publishId,e);
            //TODO 事务回滚问题
            return CompletableFuture.completedFuture("error");
        }
        return CompletableFuture.completedFuture("ok");

    }

}
